code for fun python